package com.leetcode;

public class Leetcode122 {

    //solution 1，动态规划
    //按照题意，我们可以无限次的购买，因此每一天都可能是两种状态
    //dp[i][0]代表没有股票，即当天没有股票；dp[i][1]代表持有股票，即当天持有股票。
    //我们只要将每一天的两种状态下最大的收益记录下来，并且跟前一天做比较
    public int maxProfit(int[] prices) {
        int n = prices.length;
        int[][] dp = new int[n][2];

        //第一天，没有股票收益是0
        dp[0][0] = 0;
        //第一天，持有股票，资产付出，因此收益是-prices[0]
        dp[0][1] = -prices[0];

        for (int i = 1; i < prices.length; i++) {
            //当前没有股票的状态: 前一天没有股票，或者前一天有股票但是今天卖出了
            dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][1] + prices[i]);
            //当前有股票的状态: 前一天有股票，或者前一天没有股票但是今天买入了
            dp[i][1] = Math.max(dp[i - 1][1], dp[i - 1][0] - prices[i]);
        }

        //最后一天必然是没有股票的，要么是一直没有买股票，要么是买完之后卖出
        return dp[n - 1][0];
    }


    public static void main(String[] args) {
        Leetcode122 leetcode122 = new Leetcode122();
        int[] prices = new int[]{7, 1, 5, 3, 6, 4};
        System.out.println(leetcode122.maxProfit(prices));
    }
}
